package com.study.common.util;

import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.UnsupportedEncodingException;
import java.util.List;
import java.util.Map;

/**
 * 加密过程：使用ProtostuffIOUtil把object序列化为byte数组，然后转为base64字符串，然后使用AES加密
 * 解密过程：使用AES解密，使用base64还原为byte数组，然后再次使用ProtostuffIOUtil反序列化为object
 * 
 * @author Primo Zhao
 *
 */
public class AESObjectUtil {

	public static final Logger logger = LoggerFactory.getLogger(AESObjectUtil.class);

	public static final String CHARSET = "utf-8";

	/**
	 * 加密
	 */
	public static String jsonToAESString(JSONObject obj, String encodeRules) {
		try {
			byte[] arr = obj.toJSONString().getBytes(CHARSET);
			String str = Base64Util.encodeByte(arr);
			AESPlus ap = null;
			if (encodeRules == null) {
				ap = new AESPlus();
			} else {
				ap = new AESPlus(encodeRules);
			}
			return ap.encrypt(str);
		} catch (Exception e) {
			logger.error(e.getMessage());
			return null;
		}
	}

	/**
	 * 加密
	 */
	public static String jsonToAESString(String objstr, String encodeRules) {
		try {
			byte[] arr = objstr.getBytes(CHARSET);
			String str = Base64Util.encodeByte(arr);
			AESPlus ap = null;
			if (encodeRules == null) {
				ap = new AESPlus();
			} else {
				ap = new AESPlus(encodeRules);
			}
			return ap.encrypt(str);
		} catch (Exception e) {
			logger.error(e.getMessage());
			return null;
		}
	}

	public static String jsonToAESString(Map obj, String encodeRules) {
		return jsonToAESString(JSONObject.parseObject(JSON.toJSONString(obj)), encodeRules);
	}

	/**
	 * @Description (加密json数组)
	 * @author lk
	 * @date 2019/12/6 16:57
	 * @param obj         :
	 * @param encodeRules :
	 * @return java.lang.String
	 */
	public static String jsonToAESString(JSONArray obj, String encodeRules) {
		try {
			byte[] arr = obj.toJSONString().getBytes(CHARSET);
			String str = Base64Util.encodeByte(arr);
			AESPlus ap = null;
			if (encodeRules == null) {
				ap = new AESPlus();
			} else {
				ap = new AESPlus(encodeRules);
			}
			return ap.encrypt(str);
		} catch (Exception e) {
			logger.error(e.getMessage());
			return null;
		}
	}

	/**
	 * @Description (加密传递list 进行json 调用)
	 * @author lk
	 * @date 2019/12/6 17:00
	 * @param obj         :
	 * @param encodeRules :
	 * @return java.lang.String
	 */
	public static String jsonToAESString(List obj, String encodeRules) {
		return jsonToAESString(JSONArray.parseArray(JSON.toJSONString(obj)), encodeRules);
	}

	/**
	 * 解密
	 */
	public static JSONObject toJson(String encrypted, String encodeRules) {
		try {
			AESPlus ap = null;
			if (encodeRules == null) {
				ap = new AESPlus();
			} else {
				ap = new AESPlus(encodeRules);
			}
			String str = ap.decrypt(encrypted);
			byte[] arr = Base64Util.decodeToByte(str);
			JSONObject json = JSONObject.parseObject(new String(arr, CHARSET));
			return json;
		} catch (Exception e) {
			logger.error(e.getMessage());
			return null;
		}
	}

	/**
	 * Description (重载解密方法，解密后解析成实体)
	 * 
	 * @author lk
	 * @date 2020-4-22 17:12
	 * @param encrypted   : 加密串
	 * @param encodeRules : 规则
	 * @param clazz       : 实体
	 * @return T
	 */
	public static <T> T toJson(String encrypted, String encodeRules, Class<T> clazz) {
		try {
			AESPlus ap = null;
			if (encodeRules == null) {
				ap = new AESPlus();
			} else {
				ap = new AESPlus(encodeRules);
			}
			String str = ap.decrypt(encrypted);
			byte[] arr = Base64Util.decodeToByte(str);
			T json = JSONObject.parseObject(new String(arr, CHARSET), clazz);
			return json;
		} catch (Exception e) {
			logger.error(e.getMessage());
			return null;
		}
	}

	/**
	 * @Description (解密json数组)
	 * @author lk
	 * @date 2019/12/6 16:57
	 * @param encrypted   :
	 * @param encodeRules :
	 * @return com.alibaba.fastjson.JSONArray
	 */
	public static JSONArray toJsonArr(String encrypted, String encodeRules) {
		try {
			AESPlus ap = null;
			if (encodeRules == null) {
				ap = new AESPlus();
			} else {
				ap = new AESPlus(encodeRules);
			}
			String str = ap.decrypt(encrypted);
			byte[] arr = Base64Util.decodeToByte(str);
			JSONArray json = JSONArray.parseArray(new String(arr, CHARSET));
			return json;
		} catch (UnsupportedEncodingException e) {
			logger.error(e.getMessage());
			return null;
		}
	}
}


